package com.bwt.algorithm;

import java.util.Arrays;

public class KMPAlgorithm {
    public static void main(String[] args) {
        String str1 = "BBC ABCDAB ABCDABCDABDE";
        String str2 = "ABCDABD";

        int[] next = kmpNext(str2);
        System.out.println(Arrays.toString(next));

        System.out.println(kmpSearch(str1, str2, next));
    }

    /**
     * kmp搜索算法
     * @param str1 原字符串
     * @param str2 子串
     * @param next 部分匹配表(子串匹配的部分匹配表)
     * @return -1 表示没有到 , 否则返回第一个匹配的位置
     */
    public static int kmpSearch(String str1, String str2, int[] next) {
        //遍历
        for (int i = 0,j = 0; i <str1.length(); i++) {
            //需要处理str1.charAt(i) != str2.charAt(j),调整j的大小
            //KMP算法核心点
            while (j > 0 && str1.charAt(i) != str2.charAt(j)) {
                j = next[j - 1];
            }

            if (str1.charAt(i) == str2.charAt(j)) {
                j++;
            }
            if (j == str2.length()) { //找到了
                return i - j + 1;
            }
        }
        return -1;
    }

    //获取到一个字符串 (子串) 的部分匹配表
    public static int[] kmpNext(String dest) {
        //创建一个next数组 保存部分匹配值
        int[] next = new int[dest.length()];
        //如果字符串长度为1
        next[0] = 0;
        for (int i = 1, j = 0; i < dest.length(); i++) {
            //dest.charAt(i) != dest.charAt(j) 我们需要从next[j-1] 获取新的j
            //直到我们发现有dest.charAt(i) == dest.charAt(j) 才退出
            //这个是kmp算法的核心点
            while (j > 0 && dest.charAt(i) != dest.charAt(j)) {
                j = next[j - 1];
            }

            //dest.charAt(i) == dest.charAt(j) 这个条件满足时,部分匹配值需要+1
            //i 表示前缀的末尾 j表示后缀的开头
            //如果 dest.charAt(i) == dest.charAt(j) 则表示部分匹配成功 则当前第i个元素的部分匹配表的值 等于j
            if (dest.charAt(i) == dest.charAt(j)) {
                j++;
            }
            next[i] = j;
        }
        return next;
    }
}
